from pydantic import Field
from typing import Annotated, Optional

from owl_common.base.model import AuditEntity


class GraphConfigEntity(AuditEntity):
    """
    图配置实体
    """

    # 配置id
    id: Annotated[Optional[int], Field(default=None)]

    # 配置名称
    name: Annotated[Optional[str], Field(default=None)]

    # 类型: agent，workFlow，ChatFlow
    type: Annotated[Optional[str], Field(default=None)]

    # 图配置
    graph: Annotated[Optional[dict], Field(default=None)]
